package com.lishem.top100._03slidewindow;

import java.util.ArrayList;
import java.util.List;

/**
 * https://leetcode.cn/problems/find-all-anagrams-in-a-string/description/
 * <p>
 * 给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。
 * <p>
 * 异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。
 * <p>
 * 示例 1:
 * <p>
 * 输入: s = "cbaebabacd", p = "abc"
 * <p>
 * 输出: [0,6]
 * <p>
 * 解释:
 * <p>
 * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
 * <p>
 * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
 * <p>
 * 示例 2:
 * <p>
 * 输入: s = "abab", p = "ab"
 * <p>
 * 输出: [0,1,2]
 * <p>
 * 解释:
 * <p>
 * 起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
 * <p>
 * 起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
 * <p>
 * 起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。
 */
public class _02LetCode438_找到字符串中所有字母异位词 {

    public List<Integer> findAnagrams(String s, String p) {
        char[] sChars = s.toCharArray();
        char[] pChars = p.toCharArray();
        int[] marchArray = new int[26];
        for (int i = 0; i < pChars.length; i++) {
            marchArray[pChars[i] - 'a']++;
        }
        int needMatchCount = 0;
        for (int i = 0; i < marchArray.length; i++) {
            if (marchArray[i] != 0) {
                needMatchCount++;
            }
        }
        List<Integer> res = new ArrayList<>();
        int[] window = new int[26];
        for (int i = 0; i < sChars.length; i++) {
            int curCharIndex = sChars[i] - 'a';
            window[curCharIndex]++;
            if (window[curCharIndex] == marchArray[curCharIndex]) {
                needMatchCount--;
            }
            if (needMatchCount == 0) {
                res.add(i - pChars.length + 1);
            }
            if (i >= pChars.length - 1) {
                int toDelIndex = sChars[i - pChars.length + 1] - 'a';
                window[toDelIndex]--;
                if (window[toDelIndex] == marchArray[toDelIndex] - 1) {
                    needMatchCount++;
                }
            }
        }
        return res;
    }

    public static void main(String[] args) {
        _02LetCode438_找到字符串中所有字母异位词 sol = new _02LetCode438_找到字符串中所有字母异位词();
        System.out.println(sol.findAnagrams("cbaebabacd", "abc"));
        System.out.println(sol.findAnagrams("abab", "ab"));
    }
}
